﻿// * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * *
// Copyright (c) 2011, Dr. Masroor Ehsan
// All rights reserved.
// 
// Redistribution and use in source and binary forms, with or without
// modification, are permitted provided that the following conditions are met:
//     * Redistributions of source code must retain the above copyright
//       notice, this list of conditions and the following disclaimer.
//     * Redistributions in binary form must reproduce the above copyright
//       notice, this list of conditions and the following disclaimer in the
//       documentation and/or other materials provided with the distribution.
//     * Neither the name of the author nor the
//       names of its contributors may be used to endorse or promote products
//       derived from this software without specific prior written permission.
// 
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
// ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
// WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
// DISCLAIMED. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY
// DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
// (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
// LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
// ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
// SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
// * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * *

using System.Collections.Generic;
using System.Linq;

namespace dseScraper
{
    public class QuoteFieldColumnMap
    {
        private readonly List<ColumnFieldIndex> _list;

        public QuoteFieldColumnMap()
        {
            _list = new List<ColumnFieldIndex>();
        }

        public void AddSearchPattern(StockQuoteField field, string searchPattern)
        {
            var item = FindByField(field) ??
                       new ColumnFieldIndex {Field = field, SearchPattern = searchPattern, ColumnIndex = -1};
            AddOrUpdate(item);
        }

        private void AddOrUpdate(ColumnFieldIndex item)
        {
            var entry = FindByField(item.Field);
            if (entry != null)
            {
                entry.ColumnIndex = item.ColumnIndex;
                entry.SearchPattern = item.SearchPattern;
            }
            else
            {
                _list.Add(item);
            }
        }

        public int GetColumnIndex(StockQuoteField field)
        {
            var item = FindByField(field);
            return (item != null) ? item.ColumnIndex : -1;
        }

        public int GetColumnIndex(string searchPattern)
        {
            var item = FindBySearchPattern(searchPattern);
            return (item != null) ? item.ColumnIndex : -1;
        }

        private ColumnFieldIndex FindByField(StockQuoteField field)
        {
            var item = (from c in _list
                        where (c.Field == field)
                        select c).FirstOrDefault();
            return item;
        }

        private ColumnFieldIndex FindBySearchPattern(string searchPattern)
        {
            var item = (from c in _list
                        where (c.SearchPattern == searchPattern)
                        select c).FirstOrDefault();
            return item;
        }

        public string GetSearchPatternForField(StockQuoteField field)
        {
            var item = FindByField(field);
            return (item != null) ? item.SearchPattern : string.Empty;
        }

        public StockQuoteField PriceFieldForsearchPattern(string searchPattern)
        {
            var item = FindBySearchPattern(searchPattern);
            return (item != null) ? item.Field : StockQuoteField.None;
        }

        public IEnumerable<string> GetSearchPatterns()
        {
            return _list.Select(x => x.SearchPattern);
        }

        public IEnumerable<StockQuoteField> GetPriceFields()
        {
            return _list.Where(x => x.ColumnIndex != -1).Select(x => x.Field);
        }

        public void SetColumnIndex(StockQuoteField field, int colIndex)
        {
            var item = FindByField(field) ?? new ColumnFieldIndex {Field = field};
            item.ColumnIndex = colIndex;
        }

        #region Nested type: ColumnFieldIndex

        private class ColumnFieldIndex
        {
            public int ColumnIndex;
            public StockQuoteField Field;
            public string SearchPattern;
        }

        #endregion
    }
}